# -*- coding: utf-8 -*-
from admin import *
from student import *


def login():
    print('登录')
    account = input('请输入账号:')
    pwd = input('请输入密码:')
    if account[:5] == 'admin':
        admin = Admin(account,pwd)
        print('管理员登录')
        while True:
            choose = int(input('''1.课程创建 2.创建学生学生账号 3.查看所有课程 4.查看所有学生 
                       5.查看所有学生的选课情况 6.退出程序'''))
            choose_dict = {1:admin.create_class,2:admin.create_stu_ac,3:admin.check_class,
                           4:admin.check_stu,5:admin.check_stu_choose,6:admin.exit_}
            choose_dict.get(choose)()
        